from kernel_gen.llm_api import *
from kernel_gen.question_gen.ask_questions import ask_question_for_one_chapter_pipeline
from kernel_gen.question_gen.cann_data import load_jsons_from_path

config = APIConfigFactory.get_llm_api("deepseek-r1")
setup_default_sglang_backend(config)
all_chapters = load_jsons_from_path("/home/ma-user/work/users/wty/kernel-gen/data/chapter_contents")
ask_question_for_one_chapter_pipeline(all_chapters)